#!/bin/bash

# 设置颜色
GREEN='\033[0;32m'
RED='\033[0;31m'
YELLOW='\033[1;33m'
NC='\033[0m'

# 检查Python版本
python_version=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[:2])))')
if (( $(echo "$python_version < 3.8" | bc -l) )); then
    echo -e "${RED}错误: 需要Python 3.8或更高版本${NC}"
    exit 1
fi

# 检查虚拟环境
if [ ! -d "venv" ]; then
    echo -e "${YELLOW}未找到虚拟环境，正在创建...${NC}"
    python3 -m venv venv
    source venv/bin/activate
    pip install -r requirements.txt
else
    source venv/bin/activate
fi

# 检查配置文件
if [ ! -f ".env" ]; then
    echo -e "${YELLOW}未找到.env文件，正在从示例创建...${NC}"
    cp .env.example .env
fi

# 检查Ollama服务
if ! curl -s http://localhost:11434/api/version > /dev/null; then
    echo -e "${RED}错误: Ollama服务未运行${NC}"
    echo "请先启动Ollama服务:"
    echo "ollama serve"
    exit 1
fi

# 创建必要的目录
mkdir -p logs
mkdir -p chat_sessions

# 启动应用
echo -e "${GREEN}启动Ollama Chat...${NC}"
streamlit run chat_app.py --server.port 8501 --server.address 0.0.0.0 